from rest_framework.permissions import IsAdminUser
from rest_framework.response import Response
from  rest_framework.views import APIView

from goods.models import GoodsVisitCount
from users.models import User
from datetime import date, timedelta


class UserCountView(APIView):
     # 指定管理员权限
    permission_classes = [IsAdminUser]

    """
    用户总量统计
    """
    def get(self,request):
        #获取注册用户总数（普通用户）
        count = User.objects.filter(is_staff=False).count()
        #返回数量
        return Response({'count':count})

class UserDayCountView(APIView):
     # 指定管理员权限
    permission_classes = [IsAdminUser]
    """
    当天用户总量统计
    """
    def get(self,request):
        # 获取当前日期
        now_date = date.today()
        # 获取当日活跃用户数量 date_joined 记录创建账户时间
        count = User.objects.filter(date_joined__gte=now_date).count()
        #返回数量
        return Response({'count':count})

class UserActiveCountView(APIView):

     # 指定管理员权限
    permission_classes = [IsAdminUser]
    """
    当天活跃用户总量统计
    """
    def get(self,request):
        # 获取当前日期
        now_date = date.today()
        # 获取当日登陆用户数量 last_login 记录创建账户时间
        count = User.objects.filter(last_login__gte=now_date).count()
        #返回数量
        return Response({'count':count,'date':now_date})


class UserOrderCountView(APIView):
    # 指定管理员权限
    permission_classes = [IsAdminUser]
    """
    当天活跃用户总量统计
    """

    def get(self, request):
        # 获取当前日期
        now_date = date.today()
        # 获取当日下单用户数量  orders__create_time 订单创建时间
        users = User.objects.filter(is_staff=False, orders__create_time__gte=now_date)
        #去重
        user = set(users)
        count = len(user)

        # 返回数量
        return Response({'count': count,})


class UserMonthCountView(APIView):
    # 指定管理员权限
    permission_classes = [IsAdminUser]

    def get(self, request):
        # 获取当前日期
        now_date = date.today()
        # 获取一个月前日期
        start_date = now_date - timedelta(29)
        # 创建空列表保存每天的用户量
        date_list = []

        for i in range(30):
            # 循环遍历获取当天日期
            index_date = start_date + timedelta(days=i)
            # 指定下一天日期
            cur_date = start_date + timedelta(days=i + 1)
            # 查询条件是大于当前日期index_date，小于明天日期的用户cur_date，得到当天用户量
            count = User.objects.filter(date_joined__gte=index_date, date_joined__lt=cur_date).count()

            date_list.append({
                'count': count,
                'date': index_date
            })

        return Response(date_list)


class GoodsCountView(APIView):
    """
        商品分类访问量统计
    """
    # 权限指定
    permission_classes = [IsAdminUser]

    def get(self, request):
        # 获取当天日期
        now_date = date.today()
        # 获取商品分类访问量
        goods = GoodsVisitCount.objects.filter(date__gte=now_date)
        data_list=[]
        for good in goods:
            count = good.count
            # 获取关联分类对象的名字
            category = good.category.name
            data_list.append({"count":count,'category':category})
        # 返回数量
        return Response(data_list)